package com.x.utils.video;

import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

import com.x.utils.video.model.Video;

/**
 * <p>
 * Title:VideoUtil
 * </p>
 * <p>
 * Description:
 * </p>
 * 
 * @Author Chenkangming @Date 2014-1-6
 * @version 1.0
 */
public class VideoUtil {

	public static void main(String[] args) {
		 String url = "http://v.youku.com/v_show/id_XMjU0MjI2NzY0.html";
//		 String url = "http://www.tudou.com/programs/view/xGgArUMPw9I";
//		 String url = "http://v.ku6.com/special/show_4024167/9t7p64bisV2A31Hz.html";
		// String url = "http://v.ku6.com/show/BpP5LeyVwvikbT1F.html";
		// String url = "http://6.cn/watch/14757577.html";
//		String url = "http://www.56.com/u64/v_NTkzMDEzMTc.html";
		Video video = getVideoInfo(url);
		System.out.println("视频缩略图：" + video.getPic());
		System.out.println("视频地址：" + video.getFlash());
		System.out.println("视频时长：" + video.getTime());

	}

	/**
	 * 获取视频信息
	 * 
	 * @param url
	 * @return
	 */
	public static Video getVideoInfo(String url) {
		Video video = new Video();
		try {
			if (url.indexOf("v.youku.com") != -1) {
				video = getYoukuVideo(url);
			} else if (url.indexOf("tudou.com") != -1) {
				video = getTudouVideo(url);
			} else if (url.indexOf("v.ku6.com") != -1) {
				video = getKu6Video(url);
			} else if (url.indexOf("6.cn") != -1) {
				video = get6Video(url);
			} else if (url.indexOf("56.com") != -1) {
				video = get56Video(url);
			}
		} catch (Exception e) {
			System.out.println(e);
		}
		return video;
	}

	/**
	 * 获取优酷视频
	 * 
	 * @param url
	 * @return
	 * @throws Exception
	 */
	public static Video getYoukuVideo(String url) throws Exception {
		Document doc = getURLContent(url);
		/**
		 * 获取视频缩略图
		 */
		String pic = getElementAttrById(doc, "s_sina", "href");
		int local = pic.indexOf("pic=");
		pic = pic.substring(local + 4);

		/**
		 * 获取视频地址
		 */
		String flash = getElementAttrById(doc, "link2", "value");

		/**
		 * 获取视频时间
		 */
		String time = getElementAttrById(doc, "download", "href");
		String[] arrays = time.split("\\|");
		time = arrays[4];

		Video video = new Video();
		video.setPic(pic);
		video.setFlash(flash);
		video.setTime(time);
		return video;
	}

	/**
	 * 获取土豆视频
	 * 
	 * @param url
	 * @return
	 * @throws Exception
	 */
	public static Video getTudouVideo(String url) throws Exception {
		Document doc = getURLContent(url);
		String content = doc.html();
		int beginLocal = content.indexOf("");
		content = content.substring(beginLocal);

		/**
		 * 获取视频地址
		 */
		String flash = getScriptVarByName("iid_code", content);
		flash = "http://www.tudou.com/v/" + flash + "/v.swf";

		/**
		 * 获取视频缩略图
		 */
		String pic = getScriptVarByName("thumbnail", content);

		/**
		 * 获取视频时间
		 */
		String time = getScriptVarByName("time", content);

		Video video = new Video();
		video.setPic(pic);
		video.setFlash(flash);
		video.setTime(time);
		return video;
	}

	/**
	 * 获取ku6视频
	 * 
	 * @param url
	 * @return
	 * @throws Exception
	 */
	public static Video getKu6Video(String url) throws Exception {
		Document doc = getURLContent(url);

		/**
		 * 获取视频地址
		 */
		Element flashEt = doc.getElementById("outSideSwfCode");
		String flash = flashEt.attr("value");

		/**
		 * 获取视频缩略图
		 */
		Element picEt = doc.getElementById("plVideosList");
		String time = null;
		String pic = null;
		if (picEt != null) {
			Elements pics = picEt.getElementsByTag("img");
			pic = pics.get(0).attr("src");

			/**
			 * 获取视频时长
			 */
			Element timeEt = picEt.select("span.review>cite").first();
			time = timeEt.text();
		} else {
			pic = doc.getElementsByClass("s_pic").first().text();
		}

		Video video = new Video();
		video.setPic(pic);
		video.setFlash(flash);
		video.setTime(time);
		return video;
	}

	/**
	 * 获取56视频
	 * 
	 * @param url
	 * @return
	 * @throws Exception
	 */
	public static Video get56Video(String url) throws Exception {
		Document doc = getURLContent(url);
		String content = doc.html();
		/**
		 * 获取视频缩略图
		 */
		int begin = content.indexOf("\"img\":\"");
		content = content.substring(begin + 7, begin + 200);
		int end = content.indexOf("\"};");
		String pic = content.substring(0, end).trim();
		pic = pic.replaceAll("\\\\", "");

		/**
		 * 获取视频地址
		 */
		String flash = "http://player.56.com" + url.substring(url.lastIndexOf("/"), url.lastIndexOf(".html")) + ".swf";

		Video video = new Video();
		video.setPic(pic);
		video.setFlash(flash);
		return video;
	}

	/**
	 * 获取6间房视频
	 * 
	 * @param url
	 * @return
	 * @throws Exception
	 */
	public static Video get6Video(String url) throws Exception {
		Document doc = getURLContent(url);
		/**
		 * 获取视频缩略图
		 */
		Element picEt = doc.getElementsByClass("summary").first();
		String pic = picEt.getElementsByTag("img").first().attr("src");

		/**
		 * 获取视频时长
		 */
		String time = get6VideoTime(doc, url, "watchUserVideo");
		if (time == null) {
			time = get6VideoTime(doc, url, "watchRelVideo");
		}

		/**
		 * 获取视频地址
		 */
		Element flashEt = doc.getElementById("video-share-code");
		doc = Jsoup.parse(flashEt.attr("value"));
		String flash = doc.select("embed").attr("src");

		Video video = new Video();
		video.setPic(pic);
		video.setFlash(flash);
		video.setTime(time);
		return video;
	}

	/**
	 * 获取6间房视频时长
	 * 
	 * @param doc
	 * @param url
	 * @param id
	 * @return
	 */
	private static String get6VideoTime(Document doc, String url, String id) {
		String time = null;
		Element timeEt = doc.getElementById(id);
		Elements links = timeEt.select("dt > a");
		for (Element link : links) {
			String linkHref = link.attr("href");
			if (linkHref.equalsIgnoreCase(url)) {
				time = link.parent().getElementsByTag("em").first().text();
				break;
			}
		}
		return time;
	}

	/**
	 * 获取script某个变量的值
	 * 
	 * @param name
	 *            变量名称
	 * @param content
	 * @return 返回获取的值
	 */
	private static String getScriptVarByName(String name, String content) {
		String script = content;
		int begin = script.indexOf(name);
		script = script.substring(begin + name.length() + 2);
		int end = script.indexOf(",");
		script = script.substring(0, end);
		String result = script.replaceAll("'", "");
		result = result.trim();
		return result;
	}

	/**
	 * 根据HTML的ID键及属性名，获取属性值
	 * 
	 * @param doc
	 * @param id
	 *            HTML的ID键
	 * @param attrName
	 *            属性名
	 * @return 返回属性值
	 * @throws Exception
	 */
	private static String getElementAttrById(Document doc, String id, String attrName) throws Exception {
		Element et = doc.getElementById(id);
		String arrtValue = et.attr(attrName);
		return arrtValue;
	}

	/**
	 * 获取网页内容
	 * 
	 * @param url
	 * @return
	 * @throws Exception
	 */
	private static Document getURLContent(String url) throws Exception {
		Document doc = Jsoup.connect(url).data("query", "Java").userAgent("Mozilla").cookie("auth", "token").timeout(6000).post();
		return doc;
	}
}
